# 第一个AI对话程序
import openai
import os
import dotenv
import load_dotenv

print("=== 我的第一个AI对话程序 ===")

# 加载环境变量
load_dotenv()
openai.api_key = os.getenv("OPENAI_API_KEY")


def simple_chat(message):
    """简单的聊天函数"""
    try:
        response = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=[{"role": "user", "content": message}],
            max_tokens=150,
            temperature=0.7,
        )
        return response.choices[0].message.content
    except Exception as e:
        return f"错误: {e}"


# 测试不同的提示词
test_messages = [
    "请用一句话介绍Python编程语言",
    "什么是人工智能？",
    "写一个简单的Python函数计算两个数的和",
]

print("开始与AI对话...\n")

for i, message in enumerate(test_messages, 1):
    print(f"我: {message}")
    response = simple_chat(message)
    print(f"AI: {response}")
    print("-" * 50)

print("\n🎉 恭喜！你已成功完成第一次程序化AI对话！")
